package com.gengzp.mq.server.mapper;

import com.gengzp.mq.common.model.entity.GengzpMqMessage;
import org.apache.ibatis.annotations.Mapper;
import org.apache.ibatis.annotations.Param;

import java.util.Date;
import java.util.List;

/**
 * @ClassName GengzpMqMessageMapper
 * @Description
 * @Author gengzp
 * @Date 2025/6/8 14:36
 */
@Mapper
public interface GengzpMqMessageMapper {

    /**
     * 插入一条消息
     *
     * @param message 消息
     */
    void insertMessage(GengzpMqMessage message);

    /**
     * 修改消息状态
     *
     * @param id     消息id
     * @param status 状态
     */
    void updateMessageStatus(@Param("id") Long id,
                             @Param("status") Integer status);

    /**
     * 获取一条消息
     *
     * @param id 消息key
     * @return 消息
     */
    GengzpMqMessage selectMessageByMessageKey(@Param("id") Long id);

    /**
     * 根据id查询消息集合
     *
     * @param ids id集合
     * @return 消息集合
     */
    List<GengzpMqMessage> selectMessageByIds(@Param("ids") List<Long> ids);

    /**
     * 根据状态查询消息集合
     *
     * @param status 状态
     * @return 消息集合
     */
    List<GengzpMqMessage> selectMessageByStatus(@Param("status") Integer status);

    /**
     * 查询需要重试的消息
     *
     * @return 待重试的消息集合
     */
    List<GengzpMqMessage> selectNeedRetryMessage(@Param("ids") List<Long> ids);

    /**
     * 批量修改消息状态
     *
     * @param ids    消息id集合
     * @param status 状态
     */
    void batchUpdateMessageStatus(@Param("ids") List<Long> ids,
                                  @Param("status") Integer status);

    /**
     * 批量重试消息
     *
     * @param ids 消息id集合
     */
    void batchRetryMessages(@Param("ids") List<Long> ids,
                            @Param("currentTime") Date currentTime);

    /**
     * 根据id集合批量删除消息
     *
     * @param ids 消息id集合
     */
    void batchDeleteMessageByIds(@Param("ids") List<Long> ids);

    /**
     * 根据topic批量查询消息
     *
     * @param topics topic集合
     * @return 消息集合
     */
    List<GengzpMqMessage> selectMessageByTopics(@Param("topics") List<String> topics);

    /**
     * 分页查询消息
     *
     * @param offset         偏移量
     * @param pageSize       每页数据条数
     * @param topic          主题
     * @param messageKey     消息key
     * @param messageContent 消息内容
     * @param status         消息状态
     * @return 消息集合
     */
    List<GengzpMqMessage> pageQueryMessage(@Param("offset") Integer offset,
                                           @Param("pageSize") Integer pageSize,
                                           @Param("topic") String topic,
                                           @Param("messageKey") String messageKey,
                                           @Param("messageContent") String messageContent,
                                           @Param("status") Integer status);

    /**
     * 分页查询死信队列消息
     *
     * @param offset         偏移量
     * @param pageSize       每页数据条数
     * @param topic          主题
     * @param messageKey     消息key
     * @param messageContent 消息内容
     * @param status         消息状态
     * @return 消息集合
     */
    List<GengzpMqMessage> pageQueryDeathMessage(@Param("offset") Integer offset,
                                                @Param("pageSize") Integer pageSize,
                                                @Param("topic") String topic,
                                                @Param("messageKey") String messageKey,
                                                @Param("messageContent") String messageContent,
                                                @Param("status") Integer status);

    /**
     * 查询消息计数
     *
     * @param topic          主题
     * @param messageKey     消息key
     * @param messageContent 消息内容
     * @param status         消息状态
     * @return 消息计数
     */
    Integer selectCountMessage(@Param("topic") String topic,
                               @Param("messageKey") String messageKey,
                               @Param("messageContent") String messageContent,
                               @Param("status") Integer status);

    /**
     * 查询死信消息计数
     *
     * @param topic          主题
     * @param messageKey     消息key
     * @param messageContent 消息内容
     * @param status         消息状态
     * @return 消息计数
     */
    Integer selectCountDeathMessage(@Param("topic") String topic,
                                    @Param("messageKey") String messageKey,
                                    @Param("messageContent") String messageContent,
                                    @Param("status") Integer status);

    /**
     * 根据创建时间查询
     *
     * @param createdTimeStart 创建时间-开始
     * @param createdTimeEnd   创建时间-结束
     * @return 消息集合
     */
    List<GengzpMqMessage> selectMessageByCreatedTime(@Param("createdTimeStart") Date createdTimeStart,
                                                     @Param("createdTimeEnd") Date createdTimeEnd);

    /**
     * 根据创建时间查询消费失败达到三次的消息
     *
     * @param createdTimeStart 创建时间-开始
     * @param createdTimeEnd   创建时间-结束
     * @return 消息集合
     */
    List<GengzpMqMessage> selectDeathMessageByCreatedTime(@Param("createdTimeStart") Date createdTimeStart,
                                                          @Param("createdTimeEnd") Date createdTimeEnd);

    /**
     * 根据创建截止时间查询过期消息
     * 该方法不会返回死信队列的消息，只返回正常队列的消息
     *
     * @param createdTimeEnd 创建时间-结束
     * @return 过期消息
     */
    List<GengzpMqMessage> selectTimeoutMessageByCreatedTime(@Param("createdTimeEnd") Date createdTimeEnd);

}
